"""
4.使用python实现ls命令，接受从键盘输入的文件路径，如果路径是文件夹，打印路径下有哪些文件或文件夹。如果路径是文件，就只输出当前文件本身。
"""

import os
# r的作用是防止字符串转义
file_path = r"D:\source\正则表达式30分钟入门教程-附常用表达式.chm"


# os.listdir()方法获取文件夹名字，返回数组
def get_all_files(targetDir):
    list_files = os.listdir(targetDir)
    return list_files


# 判断是否为文件
if os.path.isfile(file_path):
    print(file_path)
else:
    files = get_all_files(file_path)
    for i in files:
        print(i)


